"""
链的批处理batch
1.创建大模型对象
2.创建消息模型
3.输出解析对象StrOutputParser
4.创建连对象
5.调用连的batch方法(多消息，list(json)格式)   _chain.invoke,_chain.stream
"""
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import StrOutputParser
from model_utils import getLLM

llm = getLLM()

template = ChatPromptTemplate([
    ("system","你是一名优秀的翻译官，将用户的信息翻译成{language}"),
    ("human","{content}")
])

parser = StrOutputParser()

chain = template | llm | parser#上一组的输出是下一组的输入
msg_list = [{"language":"英语","content":"我爱你"},{"language":"日语","content":"我爱你"}]
r = chain.batch(msg_list)
print(r)

